# rocm, cuda
sudo pacman -S --noconfirm --needed rocm-hip-sdk rocm-hip-runtime hip-runtime-amd miopen-hip rocm-opencl-runtime rocm-opencl-sdk python-pytorch-rocm cuda
# cuda
sudo pacman -S --needed opencl-nvidia cuda
# opencl, vulkan
sudo pacman -S --noconfirm --needed opencl-clover-mesa lib32-opencl-clover-mesa ocl-icd opencl-headers vulkan-headers 
sudo pacman -S --noconfirm --needed opencl-rusticl-mesa lib32-opencl-rusticl-mesa ocl-icd opencl-headers pocl
# anaconda
sudo pacman -S --noconfirm --needed python-conda 
# echo "/usr/lib" >> /etc/ld.so.conf.d/00-usrlib.conf
yay -S --noconfirm hip-runtime-nvidia
ROC_ENABLE_PRE_VEGA=1

cmake ninja mkl-static
USE_FBGEMM=0 USE_CUDA=0 USE_ROCM=0 USE_XPU=0 USE_VULKAN=1 USE_VULKAN_SHADERC_RUNTIME=1 USE_VULKAN_WRAPPER=0 python setup.py install
USE_FBGEMM=0 USE_MKLDNN=0 USE_CUDA=0 USE_ROCM=0 USE_XPU=0 USE_VULKAN=1 python setup.py develop